#include "rknn_api.h"
#include <string.h>

#include <string>
#include <vector>

typedef struct{
    rknn_context rknn_ctx;
    rknn_input_output_num io_num;
    rknn_tensor_attr* input_attrs;
    rknn_tensor_attr* output_attrs;
    rknn_input* inputs;
    
    char* out_data;
}rknn_ctx;

extern std::string coco_class[80];
int model_init(rknn_ctx* rknn);
int model_inference(rknn_ctx* rknn, void* data, std::vector<float> &filterBoxes, 
                    std::vector<float> &objProbs, std::vector<int> &classId, std::vector<int> &indexArray);
